package com.ruoyi.pcj.service.impl;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.mashuai.utils.toolkit.core.msConverts.MsStreamUtils;
import com.mashuai.utils.toolkit.core.msMp.MsDb;
import com.ruoyi.basis.service.MsBaseServiceImpl;
import com.ruoyi.basis.utils.mp.MpConditionUtils;
import com.ruoyi.basis.utils.mp.MpPageUtils;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.pcj.domain.PcjAble;
import com.ruoyi.pcj.domain.PcjUserAble;
import com.ruoyi.pcj.domain.PcjUserTarget;
import com.ruoyi.pcj.dto.match.MatchUserAttrDto;
import com.ruoyi.pcj.mapper.PcjUserAbleMapper;
import com.ruoyi.pcj.service.IPcjUserAbleService;
import com.ruoyi.pcj.utils.pcj.PcjUtils;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * 擅长选项Service业务层处理
 *
 * @author ms
 * @date 2024-03-11
 */
@Service
public class PcjUserAbleServiceImpl extends MsBaseServiceImpl<PcjUserAbleMapper, PcjUserAble> implements IPcjUserAbleService {

    @Override
    public <E extends IPage<PcjUserAble>> E page(E page, Wrapper<PcjUserAble> qw) {
        return baseMapper.selectPage(page, qw);
    }

    /**
     * 查询擅长选项列表
     *
     * @param params 聊天室-查询参数 {"pageNum":1,"pageSize":10}
     * @return 擅长选项
     */
    @Override
    public IPage<PcjUserAble> selectPcjUserAbleList(Map<String, Object> params) {
        //获取条件构造器
        QueryWrapper<PcjUserAble> qw = MpConditionUtils.getQueryWrapper(params, PcjUserAble.class);
        //分页查询
        Page<PcjUserAble> page = this.page(MpPageUtils.getPage(params), qw);
        return page;
    }


    /**
     * 根据id查询擅长选项
     *
     * @param id 擅长选项主键
     * @return 擅长选项
     */
    @Override
    public PcjUserAble selectPcjUserAbleById(String id) {
        //首先获取主表信息
        PcjUserAble pcjUserAble = MsDb.getById(id, PcjUserAble.class);
        return pcjUserAble;
    }

    /**
     * 回填子表信息
     **/

    /**
     * 新增擅长选项
     *
     * @param pcjUserAble 擅长选项
     * @return 结果
     */
    @Override
    public Boolean insertPcjUserAble(PcjUserAble pcjUserAble) {
        //插入主表
        Boolean success = this.insertData(pcjUserAble);
        return success;
    }

    /**
     * 修改擅长选项
     *
     * @param pcjUserAble 擅长选项
     * @return 结果
     */
    @Override
    public Boolean updatePcjUserAble(PcjUserAble pcjUserAble) {
        //修改主表
        Boolean success = this.updateData(pcjUserAble);
        return success;
    }

    /**
     * 批量删除擅长选项
     *
     * @param ids 需要删除的擅长选项主键
     * @return 结果
     */
    @Override
    public Boolean deletePcjUserAbleByIds(String[] ids) {

        //删除主表
        Boolean success = this.removeBatchByIds(Arrays.asList(ids));
        return success;
    }

    @Override
    public List<PcjUserAble> getUserAble() {
        List<PcjUserAble> data = PcjUtils.getDataByCurUser(PcjUserAble.class);
        return data;
    }

    @Override
    public List<String> getUserAbleIds(Long userId) {
        //获取当前用户的特长
        List<PcjUserAble> userAbleList = PcjUtils.getDataByCurUserSeqNoAsc(userId, PcjUserAble.class);
        List<String> userAbleIds = MsStreamUtils.listStreamMap(userAbleList, PcjUserAble::getAbleId);
        return userAbleIds;
    }


    @Override
    public List<MatchUserAttrDto> getMatchUserAbleMsg(Long userId, Map<String, PcjAble> allAbleMap, List<MatchUserAttrDto> equalsHobbyUserList) {
        List<MatchUserAttrDto> matchUserAbleList = new ArrayList<>();
        //获取当前匹配用户的特长。包含自己特长与需要的特长，
        // 双向匹配，自己的特长可以被其他想结识的人匹配到，也可以匹配自己想结识的人
        List<PcjUserAble> userAbleList = PcjUtils.getDataByUserId(userId, PcjUserAble.class);
        if (userAbleList.size() != 0) {
            //从兴趣爱好中过滤出符合条件的用户id
            List<Long> hobbyUserIds = equalsHobbyUserList.stream().map(t -> t.getUserId()).distinct().collect(Collectors.toList());
            //获取匹配用户自己的特长id
            List<String> curUserAbleIds = userAbleList.stream().filter(t -> "0".equals(t.getType())).map(t -> t.getAbleId()).collect(Collectors.toList());
            //获取匹配用户所需xx特长的id
            List<String> curUserNeedAbleIds = userAbleList.stream().filter(t -> "1".equals(t.getType())).map(t -> t.getAbleId()).collect(Collectors.toList());
            //1.优先筛选出用户想结识xx特长的用户信息，
            LambdaQueryWrapper<PcjUserAble> qw = Wrappers.lambdaQuery();
            qw.in(PcjUserAble::getUserId, hobbyUserIds);//userId  a1,a2,a3
            qw.in(PcjUserAble::getAbleId, curUserNeedAbleIds);
            qw.eq(PcjUserAble::getType, "0");//匹配用户想结识的就是其他用户擅长的
            List<PcjUserAble> needUserList = baseMapper.selectList(qw);//a1,a2
            //如果有匹配用户想结识的特长用户
            if (needUserList.size() != 0) {
                List<Long> neeUserIds = MsStreamUtils.listStreamMap(needUserList, PcjUserAble::getUserId);
                //2.在筛选出想结识匹配用户的特长的用户id
                LambdaQueryWrapper<PcjUserAble> qw2 = Wrappers.lambdaQuery();
                qw2.in(PcjUserAble::getUserId, neeUserIds);
                qw2.in(PcjUserAble::getAbleId, curUserAbleIds);
                qw2.eq(PcjUserAble::getType, "1");//匹配用户擅长的就是其他用户想结识的
                List<PcjUserAble> otherUserNeedList = baseMapper.selectList(qw2);
                //如果此时依然有数据，达成双向匹配
                if (otherUserNeedList.size() != 0) {
                    for (PcjUserAble able : otherUserNeedList) {
                        matchUserAbleList.add(new MatchUserAttrDto(able.getUserId(), able.getAbleId()));
                    }
                }
            }
        }
        return matchUserAbleList;
    }

    @Override
    public void saveUserAble(List<PcjUserAble> list) {
        //先删除所有的，在插入
        Long userId = SecurityUtils.getUserId();
        baseMapper.realDeleteUserAble(userId);
        //重新插入
        for (PcjUserAble et : list) {
            setEntityDefValAuto(et);
        }
        this.saveOrUpdateBatch(list);
    }

/**
 *
 * 根据主表id，删除子表所有数据
 **/
    /**
     * 对外提供的设置默认字段值，id为空新增，不为空更新
     **/
    public void setEntityDefValAuto(PcjUserAble pcjUserAble) {
        this.setDefFieldVal(pcjUserAble);
    }
}
